# -*- coding: utf-8 -*-
"""
# --------------------------------------------------------
# @Author : Pan
# @E-mail : 
# @Date   : 2025-05-13 09:58:45
# @Brief  :
# --------------------------------------------------------
"""
import os

root = os.path.dirname(os.path.dirname(os.path.dirname(__file__)))
version = "0.0.1"

# TODO 特征提取模型
llm_model_zoos = {
    "MiniCPM-V-4_5":
        {"model_name": "MiniCPM-V-4_5",
         "model_file": os.path.join(root, "data/models/OpenBMB/MiniCPM-V-4_5"),
         "input_size": (224, 224),
         "max_tokens": 1024,
         "use_rgb": True,
         "describe": "MiniCPM模型",
         "download_url": "",
         },
    "Qwen3-VL-2B-Instruct":
        {"model_name": "Qwen3-VL-2B-Instruct",
         "model_file": os.path.join(root, "data/models/Qwen/Qwen3-VL-2B-Instruct"),
         "input_size": (224, 224),
         "max_tokens": 1024,
         "use_rgb": True,
         "describe": "Qwen3-VL-2B-Instruct模型",
         "download_url": "",
         },
}


# model_name = "MiniCPM-V-4_5"
model_name = "Qwen3-VL-2B-Instruct"
model_info = llm_model_zoos[model_name]
